package 每日一题;

import com.alibaba.fastjson.JSON;

import java.util.ArrayList;
import java.util.List;

/**
 * @description:
 * @author: 小白白
 * @create: 2021-10-22
 **/

public class No229求众数II {

    /**
     * 给定一个大小为 n 的整数数组，找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。
     * <p>
     * 示例 1：
     * 输入：[3,2,3]
     * 输出：[3]
     * 示例 2：
     * 输入：nums = [1]
     * 输出：[1]
     * 示例 3：
     * 输入：[1,1,1,3,3,2,2,2]
     * 输出：[1,2]
     *  
     * 提示：
     * 1 <= nums.length <= 5 * 104
     * -109 <= nums[i] <= 109
     *  
     * 进阶：尝试设计时间复杂度为 O(n)、空间复杂度为 O(1)的算法解决此问题。
     */

    public List<Integer> majorityElement(int[] nums) {

        // n / 3 最多只有两个

        // 两次摩尔投票法
        int num1 = nums[0];
        int count1 = 1;
        int num2 = Integer.MAX_VALUE;
        int count2 = 1;
        int i = 1;
        for (; i < nums.length; i++) {
            if (nums[i] != num1) {
                num2 = nums[i];
                i++;
                break;
            }
            count1++;
        }

        for (; i < nums.length; i++) {

            if (num1 == nums[i]) {
                count1++;
            } else if (num2 == nums[i]) {
                count2++;
            } else if (count1 == 0) {
                num1 = nums[i];
                count1 = 1;
            } else if (count2 == 0) {
                num2 = nums[i];
                count2 = 1;
            } else {
                count1--;
                count2--;
            }

        }

        List<Integer> result = new ArrayList<>(2);

        count1 = 0;
        count2 = 0;

        for (int num : nums) {
            if (num == num1) {
                count1++;
            }
            if (num == num2) {
                count2++;
            }
        }

        if (count1 > nums.length / 3) {
            result.add(num1);
        }

        if (count2 > nums.length / 3) {
            result.add(num2);
        }

        return result;
    }

    public static void main(String[] args) {
        No229求众数II n = new No229求众数II();
        int[] arr = {1, 2};
        List<Integer> result = n.majorityElement(arr);
        System.out.println(JSON.toJSONString(result));
    }

}
